ObjectDetection

sealed class ObjectDetection<U : InferenceModel> : OnnxModelType<U>

Object detection models and preprocessing.

Types

EfficientDetD0
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD1
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD2
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD3
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD4
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD5
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

EfficientDetD6
Link copied to clipboard

This model is a real-time neural network for object detection that detects 90 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

SSD
Link copied to clipboard
object SSD : ONNXModels.ObjectDetection<SSDObjectDetectionModel>

This model is a real-time neural network for object detection that detects 80 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2014.labels method).

SSDMobileNetV1
Link copied to clipboard

This model is a real-time neural network for object detection that detects 80 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017.labels method).

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): OnnxInferenceModel
pretrainedModel
Link copied to clipboard
abstract fun pretrainedModel(modelHub: ModelHub): U

Properties

inputShape
Link copied to clipboard
open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath
Link copied to clipboard
open override val modelRelativePath: String
preprocessor
Link copied to clipboard
open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard